# -*- coding: utf-8 -*-
# @Time    : 2016-12-07 16:00
# @Author  : wzb<wangzhibin_x@foxmail.com>
# USAGE
# python detect_blur.py --images images

# import the necessary packages
from imutils import paths
import argparse
import cv2

def variance_of_laplacian(image):
	# compute the Laplacian of the image and then return the focus
	# measure, which is simply the variance of the Laplacian
	return cv2.Laplacian(image, cv2.CV_64F).var()

# construct the argument parse and parse the arguments
#ap = argparse.ArgumentParser()
#ap.add_argument("-i", "--images", required=True,
#	help="path to input directory of images")
#ap.add_argument("-t", "--threshold", type=float, default=100.0,
#	help="focus measures that fall below this value will be considered 'blurry'")
#args = vars(ap.parse_args())

# loop over the input images
#for imagePath in paths.list_images(args["images"]):
for imagePath in paths.list_images("images"):
	# load the image, convert it to grayscale, and compute the
	# focus measure of the image using the Variance of Laplacian
	# method
	image = cv2.imread(imagePath)
	gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
	fm = variance_of_laplacian(gray)
	text = "not blurry"

	# if the focus measure is less than the supplied threshold,
	# then the image should be considered "blurry"
	#if fm < args["threshold"]:
	if fm < 100:
		text = "blurry"

	print(text)

	# show the image
	cv2.putText(image, "{}: {:.2f}".format(text, fm), (10, 30),
		cv2.FONT_HERSHEY_SIMPLEX, 0.8, (0, 0, 255), 3)
	cv2.imshow("Image", image)
	key = cv2.waitKey(0)